# -*- coding: utf-8 -*-
"""
Created on Sat May 19 15:39:56 2018

@author: Administrator
"""
# 将包导入到脚本中
import numpy as np

from matplotlib import pyplot as plt

x = np.arange(11)
y = 2 * x + 5

y2 = x**2 + 1

#plt.plot(x, y, "xb-", label = 'y1')
#plt.plot(x, y2, "+k-", label = 'y2')

# 或者
plt.plot(x, y, '^r--', x, y2, 'xb-')

plt.title("matplotlib demo")
plt.xlabel("x axis")
plt.ylabel("y axis")

plt.legend(loc = 'best')
plt.show()    # 图像显示


# 将输出图形存储为文件，默认格式为PNG，通过dpi修改输出质量
# plt.savefig('test', dpi=600) # 

# plt.plot(x, y, format_string, **kwargs)
# x:x轴数据，
# y:y轴数据
# format_string:控制曲线的格式字符串，由颜色字符、风格字符、标记字符组成
# **kwargs:更多组x,y,format_string


# plt.bar() 显示柱状图
from matplotlib import pyplot as plt 
x =  [5,8,10] 
y =  [12,16,6] 
x2 =  [6,9,11] 
y2 =  [6,15,7] 
plt.bar(x, y, align =  'center') 
plt.bar(x2, y2, color =  'g', align =  'center') 
plt.title('Bar graph') 
plt.ylabel('Y axis') 
plt.xlabel('X axis') 
plt.show()


# plt.scatter() 画散点图
from matplotlib import pyplot as plt 

data = {'a': np.arange(50),
        'c': np.random.randint(0, 50, 50),
        'd': np.random.randn(50)}
# 增加b
data['b'] = data['a'] + 10 * np.random.randn(50)
data['d'] = np.abs(data['d']) * 100

plt.scatter('a', 'b', data=data)
#plt.scatter('a', 'b', c='c', s='d', data=data)
plt.xlabel('entry a')
plt.ylabel('entry b')
plt.show()

# 小结
# 引入matplotlib包 from matplotlib import pyplot as plt
# 绘图
# plt.plot(x,y) 绘制曲线图
# plt.bar(x,y) 绘制柱状图
# 装饰图片
# plt.title("xx")
# plt.xlabel("x axis ...")
# plt.ylabel("x axis ...")
# 显示图片
# plt.show() 图像显示